from plane_sprites import *


class PlaneGame(object):
    """飞机大战游戏主类"""

    def __init__(self):
        # 游戏初始化
        # 1.创建游戏窗口
        self.screen = pygame.display.set_mode(SCREEN_RECT.size)
        # 2.创建游戏时钟
        self.clock = pygame.time.Clock()
        # 3.创建精灵组
        self.__create_sprites()
        # 4.设置定时器事件 - 创建敌机(每秒一次)
        pygame.time.set_timer(CREATE_ENEMY_EVENT, 1000)
        # 5.设置定时器事件 - 发射子弹（每0.5s一次）
        pygame.time.set_timer(FIRE_BULLET_EVENT, 500)

    def start_game(self):
        # 游戏开始
        while True:
            # 1.设置刷新帧率
            self.clock.tick(FRAME_PER_SEC)
            # 2.事件监听
            self.__event_handler()
            # 3.碰撞检测
            self.__check_collide()
            # 4.更新/绘制精灵组
            self.__update_sprites()
            # 5.更新屏幕显示
            pygame.display.update()

    def __create_sprites(self):
        # 创建精灵和精灵组
        # 创建背景精灵和精灵组
        firstBg = BackgroundSprite()
        secondBg = BackgroundSprite(True)
        self.back_ground = pygame.sprite.Group(firstBg, secondBg)
        # 创建敌机精灵组
        self.enemy_group = pygame.sprite.Group()
        # 创建英雄精灵组
        self.hero = Hero()
        self.hero_group = pygame.sprite.Group(self.hero)


    def __event_handler(self):
        # 处理事件
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                PlaneGame.__game_over()
            elif event.type == CREATE_ENEMY_EVENT:
                # 捕获到敌机事件：创建敌机并添加到精灵组
                self.enemy_group.add(EnemySprite())
            elif event.type == FIRE_BULLET_EVENT:
                self.hero.fire()
        keys_pressed = pygame.key.get_pressed()
        if keys_pressed[pygame.K_RIGHT]:
            self.hero.is_horizontal = True
            self.hero.is_vertical = False
            self.hero.speed = 2
        elif keys_pressed[pygame.K_LEFT]:
            self.hero.is_horizontal = True
            self.hero.is_vertical = False
            self.hero.speed = -2
        elif keys_pressed[pygame.K_UP]:
            self.hero.is_horizontal = False
            self.hero.is_vertical = True
            self.hero.speed = -2
        elif keys_pressed[pygame.K_DOWN]:
            self.hero.is_horizontal = False
            self.hero.is_vertical = True
            self.hero.speed = 2
        else:
            self.hero.speed = 0


    def __check_collide(self):
        # 1.检查子弹是否击中敌机类(如果击中则让子弹和敌机一起销毁)
        pygame.sprite.groupcollide(self.hero.bullets, self.enemy_group, True, True)
        # 2.检查英雄是否与敌机发生碰撞（如果发生碰撞则销毁敌机）
        enemies = pygame.sprite.spritecollide(self.hero, self.enemy_group, True)
        # 如果发生碰撞，则销毁英雄，停止游戏
        if len(enemies) > 0:
            self.hero.kill()
            PlaneGame.__game_over()

    def __update_sprites(self):
        # 更新精灵组
        # 更新背景精灵和精灵组
        self.back_ground.update()
        self.back_ground.draw(self.screen)
        # 更新敌机精灵组
        self.enemy_group.update()
        self.enemy_group.draw(self.screen)
        # 更新英雄精灵组
        self.hero_group.update()
        self.hero_group.draw(self.screen)
        # 更新子弹精灵组
        self.hero.bullets.update()
        self.hero.bullets.draw(self.screen)

    @staticmethod
    def __game_over():
        # 游戏结束
        pygame.quit()
        exit()


if __name__ == '__main__':
    # 创建游戏对象
    game = PlaneGame()
    # 启动游戏
    game.start_game()
